#!/bin/bash

# 设置主节点地址和端口
export MASTER_ADDR=localhost
export MASTER_PORT=12345

# 设置总进程数
export WORLD_SIZE=1

# 启动4个训练进程
for ((i=0; i<$WORLD_SIZE; i++))
do
    export RANK=$i
    python train/train-parallel.py &
done
wait